
__all__ = ['custom_transform', 'Refuge']
from data import Refuge
from torch.utils.data import DataLoader


def make_data(args, **kwargs):
    train_ds = Refuge.Refuge(r'./dataset', 'Training')
    valid_ds = Refuge.Refuge(r'./dataset', 'Validation')

    train_dl = DataLoader(train_ds, batch_size=args.batch_size, shuffle=True, **kwargs)
    valid_dl = DataLoader(valid_ds, batch_size=args.batch_size, shuffle=False, **kwargs)

    return train_dl, valid_dl


